/*
 *  Copyright 1999-2019 Seata.io Group.
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *       http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 */
package com.ruoyi.project.mysql.executor;

import java.sql.Statement;

import com.ruoyi.project.mysql.statement.StatementCallback;
import com.ruoyi.project.mysql.statement.StatementProxy;

/**
 * The type Plain executor.
 *
 * @author sharajava
 *
 * @param <T> the type parameter
 * @param <S> the type parameter
 */
public class PlainExecutor<T, S extends Statement> implements Executor<T> {

    private StatementProxy<S> statementProxy;

    private StatementCallback<T, S> statementCallback;

    /**
     * Instantiates a new Plain executor.
     *
     * @param statementProxy    the statement proxy
     * @param statementCallback the statement callback
     */
    public PlainExecutor(StatementProxy<S> statementProxy, StatementCallback<T, S> statementCallback) {
        this.statementProxy = statementProxy;
        this.statementCallback = statementCallback;
    }

    @Override
    public T execute(Object... args) throws Throwable {
        //其实seata是对增删改dml和select for update做了增强，只有上述sql才会走分布式事务逻辑
        //PlainExecutor是原生sql执行器，这里为了方便直接将测试增强代码写在这里了
        statementProxy.getConnectionProxy().bind("21321312");
        String beforeImg = "beforeImg";

        //执行原生sql
        T res = statementCallback.execute(statementProxy.getTargetStatement(), args);

        String afterImg = "afterImg";
        statementProxy.getConnectionProxy().appendUndoLog(beforeImg,afterImg);

        return res;
    }
}
